#include <malloc.h>
#include <stdio.h>
#include <stdlib.h>
#include <ogc/machine/processor.h>
#include <gccore.h>
#include <dirent.h>
#include <string.h>

#include "dolloader.h"

typedef struct _dolheader {
    u32 text_pos[7];
    u32 data_pos[11];
    u32 text_start[7];
    u32 data_start[11];
    u32 text_size[7];
    u32 data_size[11];
    u32 bss_start;
    u32 bss_size;
    u32 entry_point;
} dolheader;

u32 load_dol(void *dolstart, struct __argv *argv) {
    u32 i;
    dolheader *dolfile;

    if (dolstart) {
        dolfile = (dolheader *) dolstart;
        for (i = 0; i < 7; i++) {
            if ((!dolfile->text_size[i]) || (dolfile->text_start[i] < 0x100)) continue;
            VIDEO_WaitVSync();
            ICInvalidateRange ((void *) dolfile->text_start[i],dolfile->text_size[i]);
            memmove ((void *) dolfile->text_start[i],dolstart+dolfile->text_pos[i],dolfile->text_size[i]);
        }

        for (i = 0; i < 11; i++) {
            if ((!dolfile->data_size[i]) || (dolfile->data_start[i] < 0x100)) continue;
            VIDEO_WaitVSync();
            memmove ((void *) dolfile->data_start[i],dolstart+dolfile->data_pos[i],dolfile->data_size[i]);
            DCFlushRangeNoSync ((void *) dolfile->data_start[i],dolfile->data_size[i]);
        }

        memset ((void *) dolfile->bss_start, 0, dolfile->bss_size);
        DCFlushRange((void *) dolfile->bss_start, dolfile->bss_size);

        if (argv && argv->argvMagic == ARGV_MAGIC) {
            void *new_argv = (void *)(dolfile->entry_point + 8);
            memmove(new_argv, argv, sizeof(*argv));
            DCFlushRange(new_argv, sizeof(*argv));
        }

        return dolfile->entry_point;
    }
    return 0;
}
